# pip install zhipuai 请先在终端进行安装

import zhipuai

zhipuai.api_key = "41805a5412196acfdd9d63a905f9039a.Z0TWy2173NoP1ZkI"
response = zhipuai.model_api.sse_invoke(
    model="chatglm_lite",
    prompt= [{"role":"user","content":"常见知乎体框架为：\n先说一个故事吧\n自幼X，母/父/兄X，靠着X，经历了怎样的X，\n目前资产X、在X工作，说这些只想证明现在XXX，而不是XXX\n没错，故事里那个XX就是我，\n你以为就这样结束了？那之后没多久......\n请你按上文框架编写故事，给知乎问题“如何获得成功”回复，其中【xx】的内容需要你自行添加"}],
    temperature= 0.9,
    top_p= 0.7,
    incremental=True
)

for event in response.events():
    if event.event == "add":
        print(event.data, end="")
    elif event.event == "error" or event.event == "interrupted":
        print(event.data, end="")
    elif event.event == "finish":
        print(event.data)
        print(event.meta, end="")
    else:
        print(event.data, end="")